package com.shuai.utils;


import com.shuai.model.vo.ModuleVO;

import java.util.ArrayList;
import java.util.List;

public class ModuleUtils {
    public List<ModuleVO> getChildModuleList(int moduleId, List<ModuleVO> allModuleList) {
        List<ModuleVO> childModuleList = new ArrayList<>();
        for (ModuleVO moduleVO : allModuleList) {
            if (moduleVO.getParentModuleId().equals(moduleId)) {
                childModuleList.add(moduleVO);
            }
        }

        for (ModuleVO moduleVO : childModuleList) {
            List<ModuleVO> childs = getChildModuleList(moduleVO.getModuleId(), allModuleList);
            moduleVO.setChildModuleList(childs);
        }
        if (childModuleList.size() == 0) {
            return null;
        }
        return childModuleList;
    }
}